3 * Holds tests for LBFactory abstract MediaWiki class.
5 * This program is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published by
7 * the Free Software Foundation; either version 2 of the License, or
8 * (at your option) any later version.
10 * This program is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
15 * You should have received a copy of the GNU General Public License along
16 * with this program; if not, write to the Free Software Foundation, Inc.,
17 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
18 * http://www.gnu.org/copyleft/gpl.html
21 * @author Antoine Musso
22 * @copyright © 2013 Antoine Musso
23 * @copyright © 2013 Wikimedia Foundation Inc.
26 use Wikimedia\Rdbms\IDatabase
;
27 use Wikimedia\Rdbms\IMaintainableDatabase
;
28 use Wikimedia\Rdbms\LBFactory
;
29 use Wikimedia\Rdbms\LBFactorySimple
;
30 use Wikimedia\Rdbms\LBFactoryMulti
;
31 use Wikimedia\Rdbms\LoadBalancer
;
32 use Wikimedia\Rdbms\ChronologyProtector
;
33 use Wikimedia\Rdbms\MySQLMasterPos
;
34 use Wikimedia\Rdbms\DatabaseDomain
;
38 * @covers \Wikimedia\Rdbms\LBFactory
39 * @covers \Wikimedia\Rdbms\LBFactorySimple
40 * @covers \Wikimedia\Rdbms\LBFactoryMulti
42 class LBFactoryTest
extends MediaWikiTestCase
{
45 * @covers MWLBFactory::getLBFactoryClass()
46 * @dataProvider getLBFactoryClassProvider
48 public function testGetLBFactoryClass( $expected, $deprecated ) {
49 $mockDB = $this->getMockBuilder( IDatabase
::class )
50 ->disableOriginalConstructor()
54 'class' => $deprecated,
55 'connection' => $mockDB,
56 # Various other parameters required:
59 'serverTemplate' => [],
62 $this->hideDeprecated( '$wgLBFactoryConf must be updated. See RELEASE-NOTES for details' );
63 $result = MWLBFactory
::getLBFactoryClass( $config );
65 $this->assertEquals( $expected, $result );
68 public function getLBFactoryClassProvider() {
70 # Format: new class, old class
71 [ Wikimedia\Rdbms\LBFactorySimple
::class, 'LBFactory_Simple' ],
72 [ Wikimedia\Rdbms\LBFactorySingle
::class, 'LBFactory_Single' ],
73 [ Wikimedia\Rdbms\LBFactoryMulti
::class, 'LBFactory_Multi' ],
74 [ Wikimedia\Rdbms\LBFactorySimple
::class, 'LBFactorySimple' ],
75 [ Wikimedia\Rdbms\LBFactorySingle
::class, 'LBFactorySingle' ],
76 [ Wikimedia\Rdbms\LBFactoryMulti
::class, 'LBFactoryMulti' ],
81 * @covers \Wikimedia\Rdbms\LBFactory::getLocalDomainID()
82 * @covers \Wikimedia\Rdbms\LBFactory::resolveDomainID()
84 public function testLBFactorySimpleServer() {
85 global $wgDBserver, $wgDBname, $wgDBuser, $wgDBpassword, $wgDBtype, $wgSQLiteDataDir;
89 'host' => $wgDBserver,
90 'dbname' => $wgDBname,
92 'password' => $wgDBpassword,
94 'dbDirectory' => $wgSQLiteDataDir,
96 'flags' => DBO_TRX
// REPEATABLE-READ for consistency
100 $factory = new LBFactorySimple( [ 'servers' => $servers ] );
101 $lb = $factory->getMainLB();
103 $dbw = $lb->getConnection( DB_MASTER
);
104 $this->assertTrue( $dbw->getLBInfo( 'master' ), 'master shows as master' );
106 $dbr = $lb->getConnection( DB_REPLICA
);
107 $this->assertTrue( $dbr->getLBInfo( 'master' ), 'DB_REPLICA also gets the master' );
109 $this->assertSame( 'my_test_wiki', $factory->resolveDomainID( 'my_test_wiki' ) );
110 $this->assertSame( $factory->getLocalDomainID(), $factory->resolveDomainID( false ) );
112 $factory->shutdown();
116 public function testLBFactorySimpleServers() {
117 global $wgDBserver, $wgDBname, $wgDBuser, $wgDBpassword, $wgDBtype, $wgSQLiteDataDir;
121 'host' => $wgDBserver,
122 'dbname' => $wgDBname,
124 'password' => $wgDBpassword,
126 'dbDirectory' => $wgSQLiteDataDir,
128 'flags' => DBO_TRX
// REPEATABLE-READ for consistency
130 [ // emulated replica
131 'host' => $wgDBserver,
132 'dbname' => $wgDBname,
134 'password' => $wgDBpassword,
136 'dbDirectory' => $wgSQLiteDataDir,
138 'flags' => DBO_TRX
// REPEATABLE-READ for consistency
142 $factory = new LBFactorySimple( [
143 'servers' => $servers,
144 'loadMonitorClass' => LoadMonitorNull
::class
146 $lb = $factory->getMainLB();
148 $dbw = $lb->getConnection( DB_MASTER
);
149 $this->assertTrue( $dbw->getLBInfo( 'master' ), 'master shows as master' );
151 ( $wgDBserver != '' ) ?
$wgDBserver : 'localhost',
152 $dbw->getLBInfo( 'clusterMasterHost' ),
153 'cluster master set' );
155 $dbr = $lb->getConnection( DB_REPLICA
);
156 $this->assertTrue( $dbr->getLBInfo( 'replica' ), 'replica shows as replica' );
158 ( $wgDBserver != '' ) ?
$wgDBserver : 'localhost',
159 $dbr->getLBInfo( 'clusterMasterHost' ),
160 'cluster master set' );
162 $factory->shutdown();
166 public function testLBFactoryMultiConns() {
167 $factory = $this->newLBFactoryMultiLBs();
169 $dbw = $factory->getMainLB()->getConnection( DB_MASTER
);
170 $this->assertTrue( $dbw->getLBInfo( 'master' ), 'master shows as master' );
172 $dbr = $factory->getMainLB()->getConnection( DB_REPLICA
);
173 $this->assertTrue( $dbr->getLBInfo( 'replica' ), 'replica shows as replica' );
175 // Destructor should trigger without round stage errors
179 public function testLBFactoryMultiRoundCallbacks() {
181 $countLBsFunc = function ( LBFactoryMulti
$factory ) {
183 $factory->forEachLB( function () use ( &$count ) {
190 $factory = $this->newLBFactoryMultiLBs();
191 $this->assertEquals( 0, $countLBsFunc( $factory ) );
192 $dbw = $factory->getMainLB()->getConnection( DB_MASTER
);
193 $this->assertEquals( 1, $countLBsFunc( $factory ) );
194 // Test that LoadBalancer instances made during pre-commit callbacks in do not
195 // throw DBTransactionError due to transaction ROUND_* stages being mismatched.
196 $factory->beginMasterChanges( __METHOD__
);
197 $dbw->onTransactionPreCommitOrIdle( function () use ( $factory, &$called ) {
199 // Trigger s1 LoadBalancer instantiation during "finalize" stage.
200 // There is no s1wiki DB to select so it is not in getConnection(),
201 // but this fools getMainLB() at least.
202 $factory->getMainLB( 's1wiki' )->getConnection( DB_MASTER
);
204 $factory->commitMasterChanges( __METHOD__
);
205 $this->assertEquals( 1, $called );
206 $this->assertEquals( 2, $countLBsFunc( $factory ) );
207 $factory->shutdown();
208 $factory->closeAll();
211 $factory = $this->newLBFactoryMultiLBs();
212 $this->assertEquals( 0, $countLBsFunc( $factory ) );
213 $dbw = $factory->getMainLB()->getConnection( DB_MASTER
);
214 $this->assertEquals( 1, $countLBsFunc( $factory ) );
215 // Test that LoadBalancer instances made during pre-commit callbacks in do not
216 // throw DBTransactionError due to transaction ROUND_* stages being mismatched.hrow
217 // DBTransactionError due to transaction ROUND_* stages being mismatched.
218 $factory->beginMasterChanges( __METHOD__
);
219 $dbw->query( "SELECT 1 as t", __METHOD__
);
220 $dbw->onTransactionResolution( function () use ( $factory, &$called ) {
222 // Trigger s1 LoadBalancer instantiation during "finalize" stage.
223 // There is no s1wiki DB to select so it is not in getConnection(),
224 // but this fools getMainLB() at least.
225 $factory->getMainLB( 's1wiki' )->getConnection( DB_MASTER
);
227 $factory->commitMasterChanges( __METHOD__
);
228 $this->assertEquals( 1, $called );
229 $this->assertEquals( 2, $countLBsFunc( $factory ) );
230 $factory->shutdown();
231 $factory->closeAll();
233 $factory = $this->newLBFactoryMultiLBs();
234 $dbw = $factory->getMainLB()->getConnection( DB_MASTER
);
235 // DBTransactionError should not be thrown
237 $dbw->onTransactionPreCommitOrIdle( function () use ( &$ran ) {
240 $factory->commitAll( __METHOD__
);
241 $this->assertEquals( 1, $ran );
243 $factory->shutdown();
244 $factory->closeAll();
247 private function newLBFactoryMultiLBs() {
248 global $wgDBserver, $wgDBname, $wgDBuser, $wgDBpassword, $wgDBtype, $wgSQLiteDataDir;
250 return new LBFactoryMulti( [
264 'serverTemplate' => [
265 'dbname' => $wgDBname,
267 'password' => $wgDBpassword,
269 'dbDirectory' => $wgSQLiteDataDir,
270 'flags' => DBO_DEFAULT
273 'test-db1' => $wgDBserver,
274 'test-db2' => $wgDBserver,
275 'test-db3' => $wgDBserver,
276 'test-db4' => $wgDBserver
278 'loadMonitorClass' => LoadMonitorNull
::class
283 * @covers \Wikimedia\Rdbms\ChronologyProtector
285 public function testChronologyProtector() {
286 $now = microtime( true );
288 // (a) First HTTP request
289 $m1Pos = new MySQLMasterPos( 'db1034-bin.000976/843431247', $now );
290 $m2Pos = new MySQLMasterPos( 'db1064-bin.002400/794074907', $now );
293 $mockDB1 = $this->getMockBuilder( IDatabase
::class )
294 ->disableOriginalConstructor()
296 $mockDB1->method( 'writesOrCallbacksPending' )->willReturn( true );
297 $mockDB1->method( 'lastDoneWrites' )->willReturn( $now );
298 $mockDB1->method( 'getMasterPos' )->willReturn( $m1Pos );
299 // Load balancer for master DB 1
300 $lb1 = $this->getMockBuilder( LoadBalancer
::class )
301 ->disableOriginalConstructor()
303 $lb1->method( 'getConnection' )->willReturn( $mockDB1 );
304 $lb1->method( 'getServerCount' )->willReturn( 2 );
305 $lb1->method( 'hasReplicaServers' )->willReturn( true );
306 $lb1->method( 'hasStreamingReplicaServers' )->willReturn( true );
307 $lb1->method( 'getAnyOpenConnection' )->willReturn( $mockDB1 );
308 $lb1->method( 'hasOrMadeRecentMasterChanges' )->will( $this->returnCallback(
309 function () use ( $mockDB1 ) {
311 $p |
= call_user_func( [ $mockDB1, 'writesOrCallbacksPending' ] );
312 $p |
= call_user_func( [ $mockDB1, 'lastDoneWrites' ] );
317 $lb1->method( 'getMasterPos' )->willReturn( $m1Pos );
318 $lb1->method( 'getReplicaResumePos' )->willReturn( $m1Pos );
319 $lb1->method( 'getServerName' )->with( 0 )->willReturn( 'master1' );
321 $mockDB2 = $this->getMockBuilder( IDatabase
::class )
322 ->disableOriginalConstructor()
324 $mockDB2->method( 'writesOrCallbacksPending' )->willReturn( true );
325 $mockDB2->method( 'lastDoneWrites' )->willReturn( $now );
326 $mockDB2->method( 'getMasterPos' )->willReturn( $m2Pos );
327 // Load balancer for master DB 2
328 $lb2 = $this->getMockBuilder( LoadBalancer
::class )
329 ->disableOriginalConstructor()
331 $lb2->method( 'getConnection' )->willReturn( $mockDB2 );
332 $lb2->method( 'getServerCount' )->willReturn( 2 );
333 $lb2->method( 'hasReplicaServers' )->willReturn( true );
334 $lb2->method( 'hasStreamingReplicaServers' )->willReturn( true );
335 $lb2->method( 'getAnyOpenConnection' )->willReturn( $mockDB2 );
336 $lb2->method( 'hasOrMadeRecentMasterChanges' )->will( $this->returnCallback(
337 function () use ( $mockDB2 ) {
339 $p |
= call_user_func( [ $mockDB2, 'writesOrCallbacksPending' ] );
340 $p |
= call_user_func( [ $mockDB2, 'lastDoneWrites' ] );
345 $lb2->method( 'getMasterPos' )->willReturn( $m2Pos );
346 $lb2->method( 'getReplicaResumePos' )->willReturn( $m2Pos );
347 $lb2->method( 'getServerName' )->with( 0 )->willReturn( 'master2' );
349 $bag = new HashBagOStuff();
350 $cp = new ChronologyProtector(
354 'agent' => "Totally-Not-FireFox"
359 $mockDB1->expects( $this->exactly( 1 ) )->method( 'writesOrCallbacksPending' );
360 $mockDB1->expects( $this->exactly( 1 ) )->method( 'lastDoneWrites' );
361 $mockDB2->expects( $this->exactly( 1 ) )->method( 'writesOrCallbacksPending' );
362 $mockDB2->expects( $this->exactly( 1 ) )->method( 'lastDoneWrites' );
364 // Nothing to wait for on first HTTP request start
365 $cp->applySessionReplicationPosition( $lb1 );
366 $cp->applySessionReplicationPosition( $lb2 );
367 // Record positions in stash on first HTTP request end
368 $cp->storeSessionReplicationPosition( $lb1 );
369 $cp->storeSessionReplicationPosition( $lb2 );
371 $cp->shutdown( null, 'sync', $cpIndex );
373 $this->assertEquals( 1, $cpIndex, "CP write index set" );
375 // (b) Second HTTP request
377 // Load balancer for master DB 1
378 $lb1 = $this->getMockBuilder( LoadBalancer
::class )
379 ->disableOriginalConstructor()
381 $lb1->method( 'getServerCount' )->willReturn( 2 );
382 $lb1->method( 'hasReplicaServers' )->willReturn( true );
383 $lb1->method( 'hasStreamingReplicaServers' )->willReturn( true );
384 $lb1->method( 'getServerName' )->with( 0 )->willReturn( 'master1' );
385 $lb1->expects( $this->once() )
386 ->method( 'waitFor' )->with( $this->equalTo( $m1Pos ) );
387 // Load balancer for master DB 2
388 $lb2 = $this->getMockBuilder( LoadBalancer
::class )
389 ->disableOriginalConstructor()
391 $lb2->method( 'getServerCount' )->willReturn( 2 );
392 $lb2->method( 'hasReplicaServers' )->willReturn( true );
393 $lb2->method( 'hasStreamingReplicaServers' )->willReturn( true );
394 $lb2->method( 'getServerName' )->with( 0 )->willReturn( 'master2' );
395 $lb2->expects( $this->once() )
396 ->method( 'waitFor' )->with( $this->equalTo( $m2Pos ) );
398 $cp = new ChronologyProtector(
402 'agent' => "Totally-Not-FireFox"
407 // Wait for last positions to be reached on second HTTP request start
408 $cp->applySessionReplicationPosition( $lb1 );
409 $cp->applySessionReplicationPosition( $lb2 );
410 // Shutdown (nothing to record)
411 $cp->storeSessionReplicationPosition( $lb1 );
412 $cp->storeSessionReplicationPosition( $lb2 );
414 $cp->shutdown( null, 'sync', $cpIndex );
416 $this->assertEquals( null, $cpIndex, "CP write index retained" );
418 $this->assertEquals( '45e93a9c215c031d38b7c42d8e4700ca', $cp->getClientId() );
421 private function newLBFactoryMulti( array $baseOverride = [], array $serverOverride = [] ) {
422 global $wgDBserver, $wgDBuser, $wgDBpassword, $wgDBname, $wgDBprefix, $wgDBtype;
423 global $wgSQLiteDataDir;
425 return new LBFactoryMulti( $baseOverride +
[
426 'sectionsByDB' => [],
432 'serverTemplate' => $serverOverride +
[
433 'dbname' => $wgDBname,
434 'tablePrefix' => $wgDBprefix,
436 'password' => $wgDBpassword,
438 'dbDirectory' => $wgSQLiteDataDir,
439 'flags' => DBO_DEFAULT
442 'test-db1' => $wgDBserver,
444 'loadMonitorClass' => LoadMonitorNull
::class,
445 'localDomain' => new DatabaseDomain( $wgDBname, null, $wgDBprefix ),
446 'agent' => 'MW-UNIT-TESTS'
451 * @covers \Wikimedia\Rdbms\LoadBalancer::getConnection
452 * @covers \Wikimedia\Rdbms\DatabaseMysqlBase::doSelectDomain
453 * @covers \Wikimedia\Rdbms\DatabaseMysqlBase::selectDB
454 * @covers \Wikimedia\Rdbms\DatabaseMssql::selectDB
455 * @covers DatabaseOracle::selectDB
457 public function testNiceDomains() {
460 if ( wfGetDB( DB_MASTER
)->databasesAreIndependent() ) {
461 self
::markTestSkipped( "Skipping tests about selecting DBs: not applicable" );
465 $factory = $this->newLBFactoryMulti(
469 $lb = $factory->getMainLB();
471 $db = $lb->getConnectionRef( DB_MASTER
);
478 /** @var IMaintainableDatabase $db */
479 $db = $lb->getConnection( DB_MASTER
, [], '' );
484 'Null domain ID handle used'
489 'Null domain ID handle used'
494 'Main domain ID handle used; prefix is empty though'
497 $this->quoteTable( $db, 'page' ),
498 $db->tableName( 'page' ),
499 "Correct full table name"
502 $this->quoteTable( $db, $wgDBname ) . '.' . $this->quoteTable( $db, 'page' ),
503 $db->tableName( "$wgDBname.page" ),
504 "Correct full table name"
507 $this->quoteTable( $db, 'nice_db' ) . '.' . $this->quoteTable( $db, 'page' ),
508 $db->tableName( 'nice_db.page' ),
509 "Correct full table name"
512 $lb->reuseConnection( $db ); // don't care
514 $db = $lb->getConnection( DB_MASTER
); // local domain connection
515 $factory->setLocalDomainPrefix( 'my_' );
517 $this->assertEquals( $wgDBname, $db->getDBname() );
523 $this->quoteTable( $db, 'my_page' ),
524 $db->tableName( 'page' ),
525 "Correct full table name"
528 $this->quoteTable( $db, 'other_nice_db' ) . '.' . $this->quoteTable( $db, 'page' ),
529 $db->tableName( 'other_nice_db.page' ),
530 "Correct full table name"
533 $factory->closeAll();
538 * @covers \Wikimedia\Rdbms\LoadBalancer::getConnection
539 * @covers \Wikimedia\Rdbms\DatabaseMysqlBase::doSelectDomain
540 * @covers \Wikimedia\Rdbms\DatabaseMysqlBase::selectDB
541 * @covers \Wikimedia\Rdbms\DatabaseMssql::selectDB
542 * @covers DatabaseOracle::selectDB
544 public function testTrickyDomain() {
547 if ( wfGetDB( DB_MASTER
)->databasesAreIndependent() ) {
548 self
::markTestSkipped( "Skipping tests about selecting DBs: not applicable" );
552 $dbname = 'unittest-domain'; // explodes if DB is selected
553 $factory = $this->newLBFactoryMulti(
554 [ 'localDomain' => ( new DatabaseDomain( $dbname, null, '' ) )->getId() ],
556 'dbname' => 'do_not_select_me' // explodes if DB is selected
559 $lb = $factory->getMainLB();
560 /** @var IMaintainableDatabase $db */
561 $db = $lb->getConnection( DB_MASTER
, [], '' );
563 $this->assertEquals( '', $db->getDomainID(), "Null domain used" );
566 $this->quoteTable( $db, 'page' ),
567 $db->tableName( 'page' ),
568 "Correct full table name"
572 $this->quoteTable( $db, $dbname ) . '.' . $this->quoteTable( $db, 'page' ),
573 $db->tableName( "$dbname.page" ),
574 "Correct full table name"
578 $this->quoteTable( $db, 'nice_db' ) . '.' . $this->quoteTable( $db, 'page' ),
579 $db->tableName( 'nice_db.page' ),
580 "Correct full table name"
583 $lb->reuseConnection( $db ); // don't care
585 $factory->setLocalDomainPrefix( 'my_' );
586 $db = $lb->getConnection( DB_MASTER
, [], "$wgDBname-my_" );
589 $this->quoteTable( $db, 'my_page' ),
590 $db->tableName( 'page' ),
591 "Correct full table name"
594 $this->quoteTable( $db, 'other_nice_db' ) . '.' . $this->quoteTable( $db, 'page' ),
595 $db->tableName( 'other_nice_db.page' ),
596 "Correct full table name"
599 $this->quoteTable( $db, 'garbage-db' ) . '.' . $this->quoteTable( $db, 'page' ),
600 $db->tableName( 'garbage-db.page' ),
601 "Correct full table name"
604 $lb->reuseConnection( $db ); // don't care
606 $factory->closeAll();
611 * @covers \Wikimedia\Rdbms\LoadBalancer::getConnection
612 * @covers \Wikimedia\Rdbms\DatabaseMysqlBase::doSelectDomain
613 * @covers \Wikimedia\Rdbms\DatabaseMysqlBase::selectDB
614 * @covers \Wikimedia\Rdbms\DatabaseMssql::selectDB
615 * @covers DatabaseOracle::selectDB
617 public function testInvalidSelectDB() {
618 if ( wfGetDB( DB_MASTER
)->databasesAreIndependent() ) {
619 $this->markTestSkipped( "Not applicable per databasesAreIndependent()" );
622 $dbname = 'unittest-domain'; // explodes if DB is selected
623 $factory = $this->newLBFactoryMulti(
624 [ 'localDomain' => ( new DatabaseDomain( $dbname, null, '' ) )->getId() ],
626 'dbname' => 'do_not_select_me' // explodes if DB is selected
629 $lb = $factory->getMainLB();
630 /** @var IDatabase $db */
631 $db = $lb->getConnection( DB_MASTER
, [], '' );
633 \Wikimedia\
suppressWarnings();
635 $this->assertFalse( $db->selectDB( 'garbage-db' ) );
636 $this->fail( "No error thrown." );
637 } catch ( \Wikimedia\Rdbms\DBQueryError
$e ) {
638 $this->assertRegExp( '/[\'"]garbage-db[\'"]/', $e->getMessage() );
640 \Wikimedia\restoreWarnings
();
644 * @covers \Wikimedia\Rdbms\DatabaseSqlite::selectDB
645 * @covers \Wikimedia\Rdbms\DatabasePostgres::selectDB
646 * @expectedException \Wikimedia\Rdbms\DBConnectionError
648 public function testInvalidSelectDBIndependent() {
649 $dbname = 'unittest-domain'; // explodes if DB is selected
650 $factory = $this->newLBFactoryMulti(
651 [ 'localDomain' => ( new DatabaseDomain( $dbname, null, '' ) )->getId() ],
653 // Explodes with SQLite and Postgres during open/USE
654 'dbname' => 'bad_dir/do_not_select_me'
657 $lb = $factory->getMainLB();
659 if ( !$lb->getConnection( DB_MASTER
)->databasesAreIndependent() ) {
660 $this->markTestSkipped( "Not applicable per databasesAreIndependent()" );
663 /** @var IDatabase $db */
664 $lb->getConnection( DB_MASTER
, [], '' );
668 * @covers \Wikimedia\Rdbms\DatabaseSqlite::selectDB
669 * @covers \Wikimedia\Rdbms\DatabasePostgres::selectDB
670 * @expectedException \Wikimedia\Rdbms\DBExpectedError
672 public function testInvalidSelectDBIndependent2() {
673 $dbname = 'unittest-domain'; // explodes if DB is selected
674 $factory = $this->newLBFactoryMulti(
675 [ 'localDomain' => ( new DatabaseDomain( $dbname, null, '' ) )->getId() ],
677 // Explodes with SQLite and Postgres during open/USE
678 'dbname' => 'bad_dir/do_not_select_me'
681 $lb = $factory->getMainLB();
683 if ( !$lb->getConnection( DB_MASTER
)->databasesAreIndependent() ) {
684 $this->markTestSkipped( "Not applicable per databasesAreIndependent()" );
687 $db = $lb->getConnection( DB_MASTER
);
688 $db->selectDB( 'garbage-db' );
692 * @covers \Wikimedia\Rdbms\LoadBalancer::getConnection
693 * @covers \Wikimedia\Rdbms\LoadBalancer::redefineLocalDomain
694 * @covers \Wikimedia\Rdbms\DatabaseMysqlBase::selectDB
695 * @covers \Wikimedia\Rdbms\DatabaseMssql::selectDB
696 * @covers DatabaseOracle::selectDB
698 public function testRedefineLocalDomain() {
701 if ( wfGetDB( DB_MASTER
)->databasesAreIndependent() ) {
702 self
::markTestSkipped( "Skipping tests about selecting DBs: not applicable" );
706 $factory = $this->newLBFactoryMulti(
710 $lb = $factory->getMainLB();
712 $conn1 = $lb->getConnectionRef( DB_MASTER
);
715 $conn1->getDomainID()
719 $factory->redefineLocalDomain( 'somedb-prefix_' );
720 $this->assertEquals( 'somedb-prefix_', $factory->getLocalDomainID() );
722 $domain = new DatabaseDomain( $wgDBname, null, 'pref_' );
723 $factory->redefineLocalDomain( $domain );
726 $lb->forEachOpenConnection( function () use ( &$n ) {
729 $this->assertEquals( 0, $n, "Connections closed" );
731 $conn2 = $lb->getConnectionRef( DB_MASTER
);
734 $conn2->getDomainID()
738 $factory->closeAll();
742 private function quoteTable( IDatabase
$db, $table ) {
743 if ( $db->getType() === 'sqlite' ) {
746 return $db->addIdentifierQuotes( $table );
751 * @covers \Wikimedia\Rdbms\LBFactory::makeCookieValueFromCPIndex()
752 * @covers \Wikimedia\Rdbms\LBFactory::getCPInfoFromCookieValue()
754 public function testCPPosIndexCookieValues() {
756 $agentId = md5( 'Ramsey\'s Loyal Presa Canario' );
759 '3@542#c47dcfb0566e7d7bc110a6128a45c93a',
760 LBFactory
::makeCookieValueFromCPIndex( 3, 542, $agentId )
763 $lbFactory = $this->newLBFactoryMulti();
764 $lbFactory->setRequestInfo( [ 'IPAddress' => '10.64.24.52', 'UserAgent' => 'meow' ] );
766 '1@542#c47dcfb0566e7d7bc110a6128a45c93a',
767 LBFactory
::makeCookieValueFromCPIndex( 1, 542, $agentId )
772 LBFactory
::getCPInfoFromCookieValue( "5#$agentId", $time - 10 )['index'],
777 LBFactory
::getCPInfoFromCookieValue( "5@$time", $time - 10 )['index'],
782 LBFactory
::getCPInfoFromCookieValue( "0@$time#$agentId", $time - 10 )['index'],
788 LBFactory
::getCPInfoFromCookieValue( "2@$time#$agentId", $time - 10 )['index'],
793 LBFactory
::getCPInfoFromCookieValue( "2@$time#$agentId", $time +
9 - 10 )['index'],
798 LBFactory
::getCPInfoFromCookieValue( "0@$time#$agentId", $time +
9 - 10 )['index'],
799 'Almost stale; bad index'
803 LBFactory
::getCPInfoFromCookieValue( "2@$time#$agentId", $time +
11 - 10 )['index'],
809 LBFactory
::getCPInfoFromCookieValue( "5@$time#$agentId", $time - 10 )['clientId'],
814 LBFactory
::getCPInfoFromCookieValue( "5@$time#$agentId", $time +
11 - 10 )['clientId'],